Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

avx vecwise ibutterfly #368

Merged
merged 1 commit into from
Mar 3, 2024
Merged

Conversation

spapinistarkware
Copy link
Contributor

@spapinistarkware spapinistarkware commented Feb 20, 2024

This change is Reviewable

This was referenced Feb 20, 2024
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_ibutterfly branch from 604a246 to 24001a4 Compare February 22, 2024 06:32
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_vecwise_ibutterfly branch from d7d969a to 55aae90 Compare February 22, 2024 06:32
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_ibutterfly branch from 24001a4 to 131e95e Compare February 22, 2024 12:32
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_vecwise_ibutterfly branch from 55aae90 to 8dc2fed Compare February 22, 2024 12:32
This was referenced Feb 22, 2024
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_ibutterfly branch from 131e95e to d5cb3d3 Compare February 22, 2024 12:43
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_vecwise_ibutterfly branch from 8dc2fed to 8c2ac4e Compare February 22, 2024 12:43
@ilyalesokhin-starkware
Copy link
Collaborator

src/core/backend/avx512/fft.rs line 232 at r1 (raw file):

Previously, ilyalesokhin-starkware wrote…

it seems that you have only one twiddle here

ok I guess one 64 value is 2 two twiddles.

@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_vecwise_ibutterfly branch from eaab17b to 1a1a9aa Compare February 26, 2024 14:05
Copy link
Contributor Author

@spapinistarkware spapinistarkware left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Reviewable status: 0 of 1 files reviewed, 3 unresolved discussions (waiting on @ilyalesokhin-starkware)


src/core/backend/avx512/fft.rs line 197 at r1 (raw file):

    //    fft on b
    //   a:bcid
    //    fft on a

Done.


src/core/backend/avx512/fft.rs line 213 at r1 (raw file):

    //   0 1 2 3 4 5 6 7 0 1 2 3 4 5 6 7
    let t = _mm512_broadcast_i64x4(std::mem::transmute(twiddle1_dbl));
    //   Apply i:abcd => d:iabc

Done.


src/core/backend/avx512/fft.rs line 232 at r1 (raw file):

Previously, ilyalesokhin-starkware wrote…

ok I guess one 64 value is 2 two twiddles.

Done.

@ilyalesokhin-starkware
Copy link
Collaborator

src/core/backend/avx512/fft.rs line 179 at r2 (raw file):

    //   i:abcd

    // The twiddles for layer 0 are packed like:

wdyt about the following:

The twiddles for layer 0 are unique and arranged as follows: 0 1 2 3 4 5 6 7 8 9 a b c d e f.

The twiddles for layer 1 are replicated in the following pattern: 0 1 2 3 4 5 6 7 0 1 2 3 4 5 6 7.
...

Code quote:

// The twiddles for layer 0 are packed like:

@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_ibutterfly branch from 5191419 to d8211fb Compare March 3, 2024 06:42
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_vecwise_ibutterfly branch from 1a1a9aa to 755eb0c Compare March 3, 2024 06:42
Copy link
Contributor Author

@spapinistarkware spapinistarkware left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Reviewable status: 0 of 1 files reviewed, 4 unresolved discussions (waiting on @ilyalesokhin-starkware)


src/core/backend/avx512/fft.rs line 179 at r2 (raw file):

Previously, ilyalesokhin-starkware wrote…

wdyt about the following:

The twiddles for layer 0 are unique and arranged as follows: 0 1 2 3 4 5 6 7 8 9 a b c d e f.

The twiddles for layer 1 are replicated in the following pattern: 0 1 2 3 4 5 6 7 0 1 2 3 4 5 6 7.
...

Done.

Copy link
Collaborator

@ilyalesokhin-starkware ilyalesokhin-starkware left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Reviewed all commit messages.
Reviewable status: 0 of 1 files reviewed, 2 unresolved discussions (waiting on @spapinistarkware)


src/core/backend/avx512/fft.rs line 179 at r2 (raw file):

Previously, spapinistarkware (Shahar Papini) wrote…

Done.

not pushed


src/core/backend/avx512/fft.rs line 200 at r2 (raw file):

    // The twiddles for layer 2 are packed like:
    //   0 1 2 3 0 1 2 3 0 1 2 3 0 1 2 3

Suggestion:

    // The twiddles for layer 2 are replicated in the following pattern:
    //   0 1 2 3 0 1 2 3 0 1 2 3 0 1 2 3

@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_vecwise_ibutterfly branch from 755eb0c to f70e9b0 Compare March 3, 2024 08:27
Copy link
Contributor Author

@spapinistarkware spapinistarkware left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Reviewable status: 0 of 1 files reviewed, 2 unresolved discussions (waiting on @ilyalesokhin-starkware)


src/core/backend/avx512/fft.rs line 179 at r2 (raw file):

Previously, ilyalesokhin-starkware wrote…

not pushed

Done.


src/core/backend/avx512/fft.rs line 200 at r2 (raw file):

    // The twiddles for layer 2 are packed like:
    //   0 1 2 3 0 1 2 3 0 1 2 3 0 1 2 3

Done.

Copy link
Collaborator

@ilyalesokhin-starkware ilyalesokhin-starkware left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

:lgtm:

Reviewed 1 of 1 files at r3, all commit messages.
Reviewable status: :shipit: complete! all files reviewed, all discussions resolved (waiting on @spapinistarkware)

@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_ibutterfly branch from d8211fb to 62450f3 Compare March 3, 2024 09:13
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_vecwise_ibutterfly branch from f70e9b0 to 4e15286 Compare March 3, 2024 09:13
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_ibutterfly branch from 62450f3 to 8718835 Compare March 3, 2024 10:02
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_vecwise_ibutterfly branch from 4e15286 to ac428d9 Compare March 3, 2024 10:02
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_ibutterfly branch from 8718835 to bacfaff Compare March 3, 2024 11:48
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_vecwise_ibutterfly branch from ac428d9 to c6ac5cc Compare March 3, 2024 11:49
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_ibutterfly branch from bacfaff to c6032e8 Compare March 3, 2024 12:27
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_vecwise_ibutterfly branch from c6ac5cc to 3075822 Compare March 3, 2024 12:27
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_ibutterfly branch from c6032e8 to f49fb21 Compare March 3, 2024 12:56
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_vecwise_ibutterfly branch from 3075822 to 8300fe1 Compare March 3, 2024 12:56
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_ibutterfly branch 2 times, most recently from 663fc6d to ec7b594 Compare March 3, 2024 14:09
@spapinistarkware spapinistarkware changed the base branch from spapini/02-20-avx_ibutterfly to dev March 3, 2024 14:13
@spapinistarkware spapinistarkware force-pushed the spapini/02-20-avx_vecwise_ibutterfly branch from 8300fe1 to 823a462 Compare March 3, 2024 14:30
@spapinistarkware spapinistarkware merged commit 295e9ac into dev Mar 3, 2024
11 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants